import numpy as np
import os

def gen_data():
    os.makedirs("./input", exist_ok=True)
    os.makedirs("./output", exist_ok=True)

    # # 输入张量：模拟 complex64 类型
    # real_part = np.random.uniform(1, 100, [8, 2048]).astype(np.float32)
    # imag_part = np.random.uniform(1, 100, [8, 2048]).astype(np.float32)
    # input_x = real_part + 1j * imag_part   # complex64

    # # golden 输出：复数 → 全部是 0
    # golden = np.zeros(input_x.shape, dtype=np.uint8)

    # 输入张量：模拟 float16 类型
    input_x = np.random.uniform(1, 100, [8, 2048]).astype(np.float16)

    # golden 输出：因为 float16 是实数 → 全部是 1，shape 与输入一致
    golden = np.ones_like(input_x, dtype=np.uint8)

    # 保存文件
    input_x.tofile("./input/input_x.bin")
    golden.tofile("./output/golden.bin")

if __name__ == "__main__":
    gen_data()
